from jinja2 import Environment
from django.contrib.staticfiles.storage import staticfiles_storage
from django.urls import reverse


def jinja2_environment(**options):
    # jinja2环境，因为有些jinja2语法比较麻烦，比如导入静态文件，URL重定向

    # 创建环境对象
    env = Environment(**options)
    # 自定义语法{{static('静态文件')}}   {{url('路由的命名空间')}}
    env.globals.update({
        'static': staticfiles_storage.url,  # 获取静态文件的前缀
        'url': reverse,
    })

    # 返回环境对象
    return env
